class CityPage():
    def __init__(self, driver):
        self.driver = driver

    # 添加按钮
    def get_add_element(self):
        return self.driver.find_element_by_class_name('addlink')

    # 城市文本框
    def get_city_element(self):
        return self.driver.find_element_by_name('name')

    # 描述文本框
    def get_desc_element(self):
        return self.driver.find_element_by_name('desc')

    # 保存按钮
    def get_submit_element(self):
        return self.driver.find_element_by_name('_save')

    # 删除按钮
    def get_delete_element(self):
        return self.driver.find_element_by_class_name('deletelink')

    # 保存成功提示
    def get_ok_tishi_element(self):
        return self.driver.find_element_by_xpath('//*[@id="container"]/ul/li')


